Update disaggregation.py