Nonlinear finite elements/Balance of energy