clean up
create test for class
generate test for new class