Add Focus functionality to UI framework