Add extra button and flash messages