Styling, fixed flash message bug