finished logic on the train.