Dean
Fixing colab after feedback
a2c34a6