change proba function