Update collect_data.py