updated code sample in readme file