Updating TRAIN and TEST files