Python Value Iteration Vs Policy