Added trainings testing mode