1 files changed, 9 insertions, 6 deletions
diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListController.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListController.java
index 9e097ce..c636edc 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListController.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListController.java
@@ -84,12 +84,10 @@ public class EmployeeListController {
     }
 
     private void setEmployeeSelected(Employee employee, boolean selected) {
-        employeeListItemControllers.forEach(
-                employeeListItemController -> {
-                    if (employeeListItemController.getEmployee().equals(employee)) {
-                        employeeListItemController.setSelected(selected);
-                    }
-                });
+        employeeListItemControllers
+                .stream()
+                .filter(controller -> controller.getEmployee().equals(employee))
+                .forEach(controller -> controller.setSelected(selected));
     }
 
     public void selectEmployee(Employee employee) {
@@ -100,6 +98,11 @@ public class EmployeeListController {
         setEmployeeSelected(employee, false);
     }
 
+    public void deselectAllEmployees() {
+        employeeListItemControllers.forEach(
+                employeeListItemController -> employeeListItemController.setSelected(false));
+    }
+
     public static EmployeeListController createEmployeeListController(SpringFXMLLoader loader)
             throws IOException {
         FXMLWrapper<Object, EmployeeListController> wrapper =