improvements on the upgrade button