Brute force Trains
Better train representation, but still slow