UI Fixes - flash messages