Github notebook here: https://github.com/cyrusblankinship/PUI2018_cb4184/blob/master/HW8_cb4184/Assignment2_cb4184.ipynb