restart only if restart button is clicked
unify the buttons