Update example3.collapse.py