Fix close div
Merge pull request #9 from Calvin-Liono/master

Fix div error when message is passed