Finished week 2 assignment