Fixed the ipython notebook