The feature I feel would be a welcome addition to the game, would be a menu within the shop (or elsewhere) where you could select what to upgrade next.
Sometimes upgradeable things get lost in the base. This would facilitate knowing what is currently available to upgrade based on available resources.

