Add a few comments to code