added functions for creating test and training data