Added few comments to code