Modified extract_training_set. Now adds column type at each file: train, test, snIa, snII, snIbc, pec, other.