Add flash and error message styling