Add flash messages and update on styles