Tweak code snippet in readme.