making the ui nicer

main.py

@@ -40,7 +40,7 @@ class ClassSelector():
         return selected_class
 
 class PartyGenerator():
-    def __init__(self, party_size=4) -> None:
+    def __init__(self, party_size: int) -> None:
         self.size = party_size
         self.adventurers = []
         self.adventurer_types = []
@@ -63,26 +63,21 @@ class PartyGenerator():
         character_sheets = []
         for c in self.adventurers:
             character_sheets.append(c.character_sheet())
-        for i in range(15):
-            for j in range(len(self.adventurers)):
-                adv = self.adventurers[j]
-                sheet_string += adv.character_sheet()[i]
-            sheet_string += '|\n'
         return character_sheets
 
     def __str__(self):
         return f"{self.adventurers}"
 
-def returnSheets(foo):
-    new_party = PartyGenerator()
+def returnParty(party_size):
+    new_party = PartyGenerator(party_size)
     new_party.gen_party()
-    return new_party.get_character_sheets()
+    return new_party 
 
 def main():
-    character_sheets = returnSheets('foo')
-    for c in character_sheets:
-        for l in c:
-            print(f"{l}|")
+    adventurer_party = returnParty(party_size=5)
+    for adv in adventurer_party.adventurers:
+        for line in adv.vertical_sheet:
+            print(f"{line}")
         print()
 
 if __name__ == "__main__":