Update HW0.ipynb