Adding in upgrades