Update read_data.py