Add test function