Finished week 2 homework