import_data.py cleaned up