Add a test function