Python Value Iteration Bellman