worked on ipython notebook and added pickling