Improve create_patient_test_data.py