Finished week 6 assignment