Add flash messages and styling