Add test class stub