Finished week 3 assignment