Adding clean up to test script