almost done with early stopping