Tweak update.sh