add more comments to code