updated code sample in readme