Use probability distributions in context (#117)