add keyword highlighting for multinomial dirichlet and beta