Mods to week4 lab.