Code for Dirichlet MLE paper