Add a new flag to disable the graceful shutdown