add models for early stopping