Updated test script