Added comments to code for submission