updated the training structure