Update old-rebalancer.py