Update Restaurant.java

dulldesk · Jun 21, 2020 · 545d07e · 545d07e
1 parent 9ae1687
commit 545d07e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Restaurant.java b/Restaurant.java
@@ -152,6 +152,7 @@ public Restaurant(boolean training) {
 		openedStations = new HashMap<String, Boolean>();
 		openedStations.put("COVID Counter", true);
 		openedStations.put("Front Counter", true);
+		openedStations.put("Don't worry, this is my station! Go back to work!", true);
 
 
 		initialLiveTask = true;
@@ -446,7 +447,7 @@ public void paintComponent(Graphics g) {
 				}
 			} else {
 				for (Station stn : stations) {
-					if (stn.getName().equalsIgnoreCase("exit") && !User.hasTrained) {
+					if (inTraining && stn.getName().equalsIgnoreCase("exit") && !User.hasTrained) {
 						// do not allow exiting
 						continue;
 					}