Updated code sample in readme.