removed redundant code from test_mdp
wrote the estimator as well - next need to time VI solution, debug and test