remove split into training and test sets in data_preprocess
add code to split into training and test sets