Issue with Lightning-Transformed mobs (Solution)

Hello. I wanna talk about this EXTREMELY cool feature that transform mobs into others... Its called Lightning but..

Mobs that transform are often nearly killed by lightning and the fire. I think that when mob is transformed, it should not take any damage from the Lightning and get Fire Resistance for few seconds till the fire is done. This will make it much better. Thank you for your time