Update code examples in docs