Your updated suggestion was my first thought as well. That way you could follow a train as it moves from one station to the next.