Add cleanUp method for db module