Homework week 2 - added comments and week 1 feedback