Create subject_group column

Create a subject_group column to distinguish if a subject is from the
test group or the training group.