Colab Notebook

Find the colab notebook here.