Problem Set 5.ipynb