Adding cleanup func for testing