ipython notebook made for Assignment 1