Update README.md

code refinement