Update README.md
Update README.md
Some refactoring