It's pretty easy.
For Affliction, shift the damage from felpuppy into a new spell that detonates all your UA damage on the target. Maybe an aoe effect that can pop seeds.
For Demonology, shift the damage from imps into dreadstalkers, add dreadstalker charges, add them cleave, and make them only up for 5 seconds (but ofc deal the damage it previously dealt over 12 seconds), but still 15 seconds CD.
For Destruction, shift the damage from imp/immolate into chaos bolt, reduce shard cost of CB, make incinerate generate shards again but remove the RNG from immolate.
Or something like that.