Update assignment-3.ipynb