Update balancer.py