Completed functionality for one train. UI remaining