correct the ipython notebooks with the additional tests