update collect.py