Add teardown function to test example