Updated message edit handling