Rebalanced as suggested