Refactor and restyle flash messages.