Move train prompts, refactor displayStations function.