Homework for week 4 of PFAB