Styling Flash Messages + Refactoring