change business logic for subsetting training data