Hmwk 2 assignment with comments and test