Using NULL in train.c