diff --git a/Restaurant.java b/Restaurant.java index da011ff..6d50240 100644 --- a/Restaurant.java +++ b/Restaurant.java @@ -152,6 +152,7 @@ public Restaurant(boolean training) { openedStations = new HashMap(); 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; }