Added rebalancing return calculation