Update flash message classes