Fix ipython notebooks