Update collect.py