Moved upgrade menu code to Upgrades class