Fix documentation comments