Modify Prob (1) and (5)