Add documentation comments to code