Updating existing code - still beta-bernoulli distribution