UpgradeUI.cs update