minor improvements to mdp formulation