Which units would you like to replace or add?

The first thing that came to mind:

- Dwarves (because mages are not just about magic, but also about the pursuit of knowledge. Dwarves are also inventors)

- Witch Hunters (an anti-magic unit in a magic castle).

- Soldiers in armor, of the same race as the mages themselves - that is, humans. Otherwise, it's a bit strange, everyone being a mage. Maybe some don't have the gift or the money, or something else. There must be ordinary mortals...