Improve code examples in the README file