Improves test script