There is a quirk this would help a lot with -- sometimes the build screen switches back to the most basic unit (swords or slow transports) just as you are clicking on something else. It would be great to not do that, but the feature of this thread would take the pain out of it if it is not readily fixable (and I like it in general).