Programming assignment 2 added comments