worked on week 3 ipynb