added more stuff to the ipython notebook