Update README.md

just updated the code