Added rebalancing to TODO. This is made more necessary by conditional chains.