Update README.md

I get comments on code and put in README.md