Update data_io.py