change load_data.py