Add tear down function