Which units would you like to replace/add?

Instead of bone dragons, use fallen knights, or move the undead up a rank (naturally, strengthening them), and place the fallen knights in their place.