Improve code examples in README