Editing test class