Add energy flux