Update expand.py