Link to Colab notebook https://colab.research.google.com/drive/169b8g_poINzJHDU4uQgTuWSIIsendnvU?usp=sharing