Update README.md
Update comments