Update process_data.py