Interesting idea to make a puzzle where you need to manage a few units.
I want a different selection system. For example, let the player draw a boundary rectangle over the units to select them.