Updated notebooks to reflect changes in collect.py API