Modify upgrade.sh