Comments

Added comments to make the code clearer