AI upgrades