Improving the code examples in the docs