Update README.md
code refactor