Adding TVS diode