Small changes to EditDialog