Added spec for rebalancing tool