add collect.py