Added the correct changes to the ipython notebook. It is currently working