updated hw2.ipynb