Added ability to make upgrades