Refactor KeyFocusStack to be more composable.