Improved documentation generation