Btrfs: Improve space balancing code