Adding support for focus()