My problem is to many veraity of units, so im endlessly scrolling to see if just one needs an upgrade. My suggestion would be an upgrade tab or sort. The units list could really use some sorting variables. So you click it and it sorts say melee, mounted, ranged to the top of the list.
Have one of those be 'upgradeable'.