Try to sort out proper calculation of
traced energy. Still not working with chempot. Fix.