Finished up the IPython Notebook for week 5, class 1