updated UpgradeController to new version