Modified week 3 homework