Added comments to the assignment