Add cleanup of test files