Update the code sample in readme.md