Add test for Bernoulli