Updated main test script