done

divide train and test data