Put in some code
Move computation into script driven by ipynb