Better support for flash messages