Add test class