Better implementation of display-message widget