Update data_collect.py