HW 4 Part 1. ipynb