train controller ui slight improvement