Updated the test script