Added function to return map of probabilities for debugging