Update trainCtrl.c