Fix message ui bug
Merge pull request #37 from windyarts/master

UI bug fix