Improve code examples in readme.