more work on the training