Update pset0.ipynb