separate train and test group