added HW4.ipynb