add probability for first stage, fixed