Optimise test script