WIP: assign features to trains.