Added output of nonlinear term energy for SS and VE models