Updated code sample in README