Problem Set 2.ipynb