modified test class