Improve MessageController

Improve MessageController