complete hw1.ipynb