added restart prompt at the end