added a test class and header file