Bugfix
Updates due to new function for calculating probabilities for whole trajectory