Added line to update the probability vector.