updated something here for the training