Idea to do this in an ipynb. Run with $ipython notebook in this folder.