1 files changed, 16 insertions, 5 deletions

diff --git a/src/jcgp/gui/GUI.java b/src/jcgp/gui/GUI.java
index e7a4a23..91c72e6 100644
--- a/src/jcgp/gui/GUI.java
+++ b/src/jcgp/gui/GUI.java
@@ -40,6 +40,8 @@ public class GUI extends Application {
     
 	private final JCGP jcgp;
 	
+	private Stage stage;
+	
 	private final FunctionSelector functionSelector;
 	
 	private PopulationPane populationPane;
@@ -108,7 +110,7 @@ public class GUI extends Application {
 	public void start(Stage primaryStage) throws Exception {
 		console = new GUIConsole();
 		jcgp.setConsole(console);
-		
+		stage = primaryStage;
 		/*
 		 * Instantiate the various GUI elements here. 
 		 * 
@@ -181,15 +183,20 @@ public class GUI extends Application {
 	
 	public void reset() {
 		if (!running && settingsPane.areParametersValid()) {
+			setEvaluating(false);
 			settingsPane.applyParameters();
 			jcgp.reset();
-			populationPane.remakeTabs();
-			settingsPane.revalidateParameters();
-			settingsPane.updateControls(false, jcgp.isFinished());
-			console.flush();
+			reDraw();
 		}
 	}
 	
+	public void reDraw() {
+		populationPane.remakeTabs();
+		settingsPane.revalidateParameters();
+		settingsPane.updateControls(false, jcgp.isFinished());
+		console.flush();
+	}
+	
 	private void runningMode(boolean value) {		
 		if (value) {
 			populationPane.unlockOutputs();
@@ -234,4 +241,8 @@ public class GUI extends Application {
 	public void setEvaluating(boolean value) {
 		populationPane.setEvaluating(value);
 	}
+
+	public Stage getStage() {
+		return stage;
+	}
 }