I think it would be a good idea to be able to organize your units in different ways.
For example, saving certain "squads" containing units with specific loadouts for easy use would speed up most people's combat significantly. Being able to rename and reorder units would also assist with organization.
Also, rather than just viewing units by type, maybe there could be different filters so if you only wanted to see the HI you could, or if you wanted the units to be sorted by level rather than type you could.
Right now the units page is pretty cluttered and disorganized, and it makes it a bit harder than it needs to be to set up for combat.