Small value_iteration imrpovement