Ideas and suggestions.

Regarding the issue of water magic, which is significantly weaker than other schools. In the first and second parts of the game, there was a spell called Storm. Mechanically, it was the same as Armageddon, but weaker. It's unclear why it was removed. If this spell is reintroduced, as a weaker version of Armageddon, at level 2, the combination of Storm + Water Protection (+ mass healing, especially on bulky tier 1 units) could be quite effective. This would encourage players to use water-based units in certain situations, which would increase the variety in battles. I believe it's important to emphasize that this spell is a canonical element from the game series, not something invented by an unknown person. I hope that creating the animation for it won't be too difficult.