Trying to improve test script