Add styling to flash messages