Add documentation generation and code comments