HW2.ipynb