Refactor Balancer into an interface