Not running value iteration if only 1 state (then it does not converge)