updated comments to programming assignment 2