Focus input on edit