Value Function Iteration fixes