#18 
Probability control is added