Added some comments to code