Add support for multiple trains to the track controller