Update decompose.py