Working on ML train control.