Update for week 2 homework.