Adjust code examples in documentation