Selection demo is broken due to 3.x.x focus differences.

[bonafideduck]

Problem:

When the Selection demo is run in releases 2.x.x, it works, but not in 3.x.x. The difference is that 3.x.x loses focus when the buttons are clicked.

Action required:

Determine if the focus difference is correct and either fix it if not or fix the Selection demo to handle the change.