Added a teardown function