Improve styling and displaying of the messages