Add attributes to split test set