Added some upgrades