add prob soft logic