Refactor MessageEntity
Merge