Add flash message support.