Updated the class docs