update comments in code based on README.md