Address code review comments
Add rebalance new cluster script