Update train.c

Formatting changes