Update pset1.ipynb