Changed code sample in README to match screenshot