add classify to train