Add functions to show next train