Update UI of edit message