HW 1.ipynb