Implement balancing 1N and add a few tests