more dataset and test/training data stuff