Added optimize and diagnose to test_bernoulli