Update README.md

Updated code