Use bernoulli instead of binomial distribution where appropriate