It'd be even cooler if the train came with a track brick pack that the train would follow.
I cannot model, and following a track is a bit difficult.
I can make it so when it's stopped, it will search around for the nearest track brick and align, but aligning while it's moving is impractical.