Update thermal energy calc so that it never exceeds initial PE

[#81553958]
Tweak layout to allow more room for the model

[#81553958]