1 files changed, 7 insertions, 8 deletions

diff --git a/src/main/java/space/m0e/quga/oop/lab56maven/dialogs/InsertDialog.java b/src/main/java/space/m0e/quga/oop/lab56maven/dialogs/InsertDialog.java
index b4aba1d..a3d9425 100644
--- a/src/main/java/space/m0e/quga/oop/lab56maven/dialogs/InsertDialog.java
+++ b/src/main/java/space/m0e/quga/oop/lab56maven/dialogs/InsertDialog.java
@@ -20,10 +20,7 @@ import  space.m0e.quga.oop.lab56maven.entities.micro.Nobel;
 import java.awt.*;
 import java.io.IOException;
 import java.net.URL;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Objects;
-import java.util.ResourceBundle;
+import java.util.*;
 
 import static space.m0e.quga.oop.lab56maven.entities.common.Utils.sendErrorAlert;
 
@@ -92,27 +89,29 @@ public class InsertDialog implements Initializable {
             case "Immigrant" -> {
                 Immigrant immigrant = new Immigrant(firstName.getText(), lastName.getText(), inputAge, inputX, inputY, inputAbility);
                 if (!generateItems.isSelected()) {
-                    immigrant.setItems(new ArrayList<>());
+                    immigrant.setItems(new HashMap<>());
                 }
                 Main.immigrants.add(immigrant);
-                immigrant.getThread().start();
+                immigrant.getTimeline().play();
 
             }
             case "Dwarf" -> {
                 Dwarf dwarf = new Dwarf(firstName.getText(), lastName.getText(), inputAge, inputX, inputY, inputAbility);
                 if (!generateItems.isSelected()) {
-                    dwarf.setItems(new ArrayList<>());
+                    dwarf.setItems(new HashMap<>());
                 }
 
+                dwarf.getTimeline().play();
                 Main.immigrants.add(dwarf);
             }
 
             case "Nobel" -> {
                 Nobel nobel = new Nobel(firstName.getText(), lastName.getText(), inputAge, inputX, inputY, inputAbility);
                 if (!generateItems.isSelected()) {
-                    nobel.setItems(new ArrayList<>());
+                    nobel.setItems(new HashMap<>());
                 }
 
+                nobel.getTimeline().play();
                 Main.immigrants.add(nobel);
             }
         }