Improve UI with focus()