added restarttraining message