Print statements in training method