Improved class documentation