Updated class documentation