Adding some comments to code