Small update on the train