Fixed the documentation comments