Problem Set 1.ipynb