Update README.md with comments on code