added train test data