Updates README.md and adds comments