Changes to test script