Update train.c