Подскажите как раскидать классы?

 
import random 
 
class Elf: 
    def __init__(self, name): 
        self.name = name 
 
class Tree: 
    def __init__(self, branches): 
        self.branches = branches 
        self.homes = [[] for _ in range(branches)] 
 
    def populate_homes(self): 
        for i in range(self.branches): 
            print(f"Дом {i + 1}:") 
            while True: 
                elf_name = input("Введите имя эльфа (или 'None' для пропуска): ") 
                if elf_name.lower() == "none": 
                    break 
                elf = Elf(elf_name) 
                self.homes[i].append(elf) 
 
    def find_elf_neighbors(self, elf_name): 
        total_neighbors = 0 
        for home in self.homes: 
            for elf in home: 
                if elf.name == elf_name: 
                    total_neighbors += len(home) - 1  # исключаем самого эльфа 
                    break 
        return total_neighbors 
 
# Создаем деревню с пятью деревьями и случайным количеством веток на каждом дереве 
tree_count = 5 
trees = [Tree(random.randint(3, 5)) for _ in range(tree_count)] 
 
# Располагаем эльфов в домах 
for tree in trees: 
    tree.populate_homes() 
 
# Находим количество соседей для определенного эльфа 
elf_name_to_find = input("Введите имя эльфа для поиска соседей: ") 
total_neighbors_found = 0 
for tree in trees: 
    total_neighbors_found += tree.find_elf_neighbors(elf_name_to_find) 
 
print(f"Количество соседей для эльфа {elf_name_to_find}: {total_neighbors_found}")