revise the train user-interface