Updated code in README.md