Adding in the probabilities.  Hardcoded at the moment.