Improve documentation comments