Updates to test script