add probabilities, not working yet