Adding some comments to the code