flash messages working better