Improve documentation generation