Programming Assignment 2 with code comments