Added logic to clean up after the deployment