Edit to week 6 homework write up