Tidying up test script.