Corrections to week3.ipynb