removed global usage of TRAIN_DATA_DIR