minor update for colab session