Add focus on edit