Little changes for Flash messages