This is for the training class