updated create_test_data_all.py with imports