Improve generated documentation handling