fix(ui): More sane focus management

Use the focus stack rather than setting focused items manually.