Add energy flux to exercise-basic

When switching to non-isothermal there should be also an energy flux for injecting Nitrogen.