minor cleanup in collections_demo.py