Fix code sample in readme