Add the test function