Dirichlet model is now a HashMap instead of 3D array