Focus on the input element when editing