Add some comments to code