Begin adding functionality to upgrades