The jupiter notebook is in the repo. But I have also attached the link to the colab to run it if needed. Link to the colab: https://colab.research.google.com/drive/1YAdtLLma2UMZVlW0c6RK-RnHwguufzUs?usp=sharing