Working on method for determining which train to take