Homework for week 4