Update focus for editable elements