Finishing Week 3 homework