Update import_data script