separated training from test data