Update colab.py