Got flash messages working