add some comments
finish week 7 homework