HW 5.ipynb