Updated README.md and added comments