Modifying the test script (is now a python script)