Program terminates now after saying no to a new game

src/main/java/com/example/tictactoeswtha2/Controller.java

@@ -78,10 +78,10 @@ public class Controller {
             Button noBtn =  new Button("No");
             yesBtn.setOnAction(e->{
                 restart();
-                System.exit(0);
+                stage.close();
                     });
             noBtn.setOnAction(e->{
-                stage.close();
+                System.exit(0);
             });
             VBox layout = new VBox(10);
             layout.getChildren().addAll(winnerLabel, label, yesBtn, noBtn);