add probability of ending