expand.py updated