Mods to week4.ipynb