work on trainingMode