data split into training and test data