add updated train controls ui