Modify the comments in rebalance-fun