Added water limitation back to allocation routinue