changes to test class