got rid of test class