added a test class