Fixing doc code examples