finish tests and class for hero