update train controller prototype
update for train controller prototype