Added ipython notebook testing code