Updated documentation

Updated the class documentation.