Problem Set 4.ipynb