Updates to week2 lab.