Add more comments to code