Energy balance equation