ipython notebook for problem 1.1