Be nice if the ability to just click the upgrades on the training screen would auto buy the number of units you need for that upgrade, aslong as you have enough KI obviously :)
maybe some hotkeys would make gameplay nicer instead of having to click each different tab.. and things like craft and sell can be awkward if you are doing it often