Add the test script