Upgrade Script additions