Balancing in Apache Beam