add method to train controller