added HW2.ipynb