Update data_transfer.py