tidied up test file