updated the ipython notebook for review