WIP - breaking everything to upload for remote colab.