Heres a compromise, the dragons a hero style unit, BUT, the baby dragon can choose its way, and morph into any of the dragon types which would be green dragon, ice dragon, red dragon etc etc, second of all there could be a series of minor dragons that are weak and roughly the equal of magic dragons, the wyrm could be a ground hero unit, while the dragon would be a hero air unit.