Updated Upgrade Controller