Update refine_data.py