Finished off week 2 homework