Added edit message functionality.