adding test class