Fix documentation in comments