Add class documentation