multinomial and dirichlet stuff working