Update run_upgrade.sh