edited the test classes