Updating shut down logic