"Restart" button added