Update train.h