Update train.c
Merge pull request #1 from Dust1n/master

Update train.c