Add new test class