Update README.md

Updated the sample code to make it working