I think it would be really amazing if the heroes keep their own unique attacks with only a theme change when they equip an orb.
Example: If the Keeper of the Grove gets a fire orb he should still attack with his “magic bird attack”, but it should be red instead of green.
Also, get rid of the fire effects from the hydromancer. These are small things, but it makes the units better overall (imo).