Refactor flash messages to partial in user