Updated sample code on README.md