Add test utility class