Update W5 Assignment.ipynb