Changed the example data set
removed unnecessary imports
Adding an IPython notebook as usage example