Upgrade Controller Implementation