Working on edit dialog