fix documentation
fix class
fix class