add unit test class