Add commentary to the ipython notebook, clean up some things