Updated README.md and comments