Added upgrades functionality