I am putting together some helper functions to make it easier to work in Colab. Take a look. I'm looking forwards to any suggestions.