Improved rebalance function in neutron-ha-tool

Improved rebalance function in neutron-ha-tool