Update assignment-4.ipynb