add prob logic