Add styling for flash messages