Add message class to flash messages