Improve documentation of class