Added prompting for upgrades