Updated code in readme.md