added probabilistic energy loss