Dirichlet modelling for MDP is working, but somewhat buggy.