add a popup + button at end to restart