Update assignment3.ipynb