expand on calling functions in colab https://stackoverflow.com/questions/47345004/in-googles-colab-notebook-how-do-i-call-a-function-from-a-python-file?answertab=votes#tab-top