Working on week 3 homework