diff --git a/app/main.py b/app/main.py
index e1e6ad0e2..b7bc1f470 100644
--- a/app/main.py
+++ b/app/main.py
@@ -11,15 +11,15 @@ def __str__(self) -> str:
 
 
 def create_person_list(people: list) -> list:
-    person_list = []
+    person_list = [
+        Person(name=person["name"], age=person["age"])
+        for person in people
+    ]
 
     for index, person in enumerate(people):
-        person_list.append(Person(name=person["name"], age=person["age"]))
-
-    for index, person in enumerate(people):
-        if "wife" in person and person["wife"] is not None:
+        if person.get("wife"):
             person_list[index].wife = Person.people[person["wife"]]
-        elif "husband" in person and person["husband"] is not None:
+        elif person.get("husband"):
             person_list[index].husband = Person.people[person["husband"]]
 
     return person_list