Working prediction of transitions and rewards

TODO: make planner work
mountain car testing