Added modifications to the ipython notebook