cleaned up ipynb a bit