Add in AvailabilityZone balancing