I think it's time to add a lightning rod. Add it's functionality to tesla towers or add a new trap. I would basiclly trigger if a lightning spell i used over it, absorbing all or most of the damage done. TT could receive 3x damage or a new trap would break after one use.
As for balance I have faith SC can make it work one way or another. They've done right by me thus far.

