Update Assignment 2.ipynb