Problem Set 6.ipynb