Update Pset1.ipynb