Update control_train.ino