Refactoring du code pour inclure le message flash