Changes to unit testing Ipython Notebook