Docs: updated code sample