Added a simplified version of the colab notebook, which doesnt have a clean env setup. This simplifies the setup process greatly at the cost of hurting reproducibility. Further research into ways to get a clean env on colab is required