Small value_iteration improvement