Improved Policy print and test code. Can extract policy from Q function in MDP.