Tweak the code examples in the readme