tidied up test script
(#401)