Updates for week 4 homework