Upgrade Script functionality