just for the training