add documentation for class