Updating class documentation