Add close feature to flash message