The Colab link: https://colab.research.google.com/drive/19TGZMiEJ4UUmHC6oHLmV0iBAMn4X4F73?usp=sharing