Added some class documentation