diff --git a/cv7/program.c b/cv7/program.c
index 918b3f4..7b64a25 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -1,132 +1,120 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <assert.h>
 
 #define SIZE 256
 
-// Struktúra uzla
-typedef struct tree {
-    char value[SIZE];
-    struct tree* left;
-    struct tree* right;
-} Tree;
+// Структура для вузла дерева
+struct tree {
+    char value[SIZE];  // Текст питання або твердження
+    struct tree *left; // Лівий підвузол
+    struct tree *right; // Правий підвузол
+    int id; // Ідентифікатор вузла
+};
 
-// Funkcia na vytvorenie nového uzla
-Tree* create_node(const char* value) {
-    Tree* node = malloc(sizeof(Tree));
-    if (node) {
-        strncpy(node->value, value, SIZE);
-        node->left = NULL;
-        node->right = NULL;
-    }
-    return node;
+// Функція для знищення дерева та звільнення пам'яті
+void destroy_tree(struct tree *tree) {
+    if (tree == NULL) return;  // Якщо вузол порожній, нічого не робимо
+    destroy_tree(tree->left);  // Рекурсивно знищуємо лівий підвузол
+    destroy_tree(tree->right); // Рекурсивно знищуємо правий підвузол
+    free(tree);  // Звільняємо пам'ять, виділену для поточного вузла
 }
 
-// Funkcia na načítanie stromu
-Tree* read_tree() {
-    char buffer[SIZE];
-    memset(buffer, 0, SIZE);
-
-    // Čítanie riadku
-    if (!fgets(buffer, SIZE, stdin)) {
-        return NULL;
-    }
-
-    // Odstránenie znaku nového riadku
-    buffer[strcspn(buffer, "\n")] = 0;
-
-    if (buffer[0] == '\0') {
-        return NULL; // Prázdny riadok znamená koniec
-    }
-
-    Tree* node = create_node(buffer);
-
-    // Ak ide o odpoveď, nečítame podstromy
-    if (buffer[0] == '*') {
-        return node;
-    }
-
-    // Čítanie podstromov (ľavý a pravý uzol)
-    node->left = read_tree();
-    node->right = read_tree();
-    return node;
+// Функція для підрахунку кількості листових елементів (відповідей) в дереві
+int count_items(struct tree *tree) {
+    if (tree == NULL) return 0;  // Якщо вузол порожній, повертаємо 0
+    if (tree->left == NULL && tree->right == NULL) return 1;  // Якщо вузол є листом (не має підвузлів), то це відповідь
+    return count_items(tree->left) + count_items(tree->right);  // Рекурсивно підраховуємо кількість листових елементів
 }
 
-// Funkcia na zničenie stromu
-void destroy_tree(Tree* tree) {
-    if (tree) {
-        destroy_tree(tree->left);
-        destroy_tree(tree->right);
-        free(tree);
+// Основна функція для запуску системи рішень
+void run_system(struct tree *node) {
+    if (node == NULL) return;  // Якщо вузол порожній, припиняємо виконання
+
+    int empty_lines = 0;  // Лічильник порожніх рядків
+    int first_prompt = 1; // Флаг для виведення першого підказки
+
+    // Основний цикл, який буде працювати, поки є вузли в дереві
+    while (node != NULL) {
+        // Якщо вузол є листом (в кінці дерева)
+        if (node->left == NULL && node->right == NULL) {
+            printf("%s", node->value);  // Виводимо відповідь
+            printf("Koniec\n");  // Завершуємо програму
+            return;
+        }
+
+        // Перший запит до користувача, вибір між двома варіантами
+        if (first_prompt) {
+            printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+            first_prompt = 0;  // Встановлюємо флаг, щоб підказка виводилась тільки один раз
+        }
+
+        printf("%s", node->value);  // Виводимо питання
+
+        char answer;
+        int result = scanf(" %c", &answer);  // Читаємо відповідь користувача
+
+        // Перевірка на кінець вводу або два порожні рядки поспіль
+        if (result == EOF || (answer == '\n' && ++empty_lines >= 2)) {
+            printf("Koniec vstupu\n");
+            return;  // Завершуємо виконання програми
+        }
+
+        // Обробка відповіді користувача
+        if (answer == 'a') {
+            empty_lines = 0;  // Скидаємо лічильник порожніх рядків
+            node = node->left;  // Переходимо до лівого підвузла
+        } else if (answer == 'n') {
+            empty_lines = 0;  // Скидаємо лічильник порожніх рядків
+            node = node->right;  // Переходимо до правого підвузла
+        } else {
+            printf("Nerozumiem\n");  // Якщо відповідь неправильна
+            return;  // Завершуємо програму
+        }
     }
 }
 
-// Funkcia na preorder výpis stromu
-void print_tree(Tree* tree, int offset) {
-    if (tree == NULL) return;
-    for (int i = 0; i < offset; i++) {
-        printf(" ");
-    }
-    printf("%s\n", tree->value);
-    print_tree(tree->left, offset + 3);
-    print_tree(tree->right, offset + 3);
-}
-
-void ask_question(Tree* tree) {
-    if (!tree) return;
-
-    printf("%s (a/n) ?\n", tree->value);
-
-    // Ak ide o odpoveď, ukončujeme
-    if (tree->left == NULL && tree->right == NULL) {
-        printf("Koniec\n");
-        return;
+// Функція для читання дерева з вводу
+struct tree* read_tree(int *counter) {
+    char buffer[SIZE];  // Буфер для зберігання рядка
+    if (fgets(buffer, SIZE, stdin) == NULL || buffer[0] == '\n') {
+        return NULL; // Якщо рядок порожній або не вдалося прочитати, повертаємо NULL
     }
 
-    char answer;
-    scanf(" %c", &answer);
-
-    // Kontrola platnej odpovede
-    if (answer == 'a') {
-        ask_question(tree->left); // Pre odpoveď "a"
-    } else if (answer == 'n') {
-        ask_question(tree->right); // Pre odpoveď "n"
-    } else {
-        printf("Neplatná odpoveď, koniec.\n");
+    struct tree *node = calloc(1, sizeof(struct tree));  // Створюємо новий вузол
+    if (node == NULL) {
+        return NULL; // Якщо пам'ять не вдалося аллокувати, повертаємо NULL
     }
+
+    strcpy(node->value, buffer);  // Копіюємо значення в поле value
+    node->id = (*counter)++;  // Присвоюємо унікальний ідентифікатор вузла
+
+    // Якщо це не лист, то читаємо лівий та правий підвузли
+    if (buffer[0] != '*') {  
+        node->left = read_tree(counter);  // Рекурсивно читаємо лівий підвузол
+        node->right = read_tree(counter); // Рекурсивно читаємо правий підвузол
+    }
+
+    return node;  // Повертаємо створений вузол
 }
 
 int main() {
-    // Načítanie stromu zo vstupu
-    Tree* root = read_tree();
+    int counter = 0;  // Лічильник для присвоєння ідентифікаторів вузлам
+    struct tree *root = read_tree(&counter);  // Читаємо дерево з вводу
+
+    printf("Expert z bufetu to vie.\n");
 
-    // Ak strom nie je prázdny
     if (root == NULL) {
-        printf("Chyba pri načítaní pravidiel.\n");
-        return 1;
+        printf("Chybna databaza\n");
+        return 0;  // Якщо дерево порожнє, завершуємо виконання
     }
 
-    // Preverenie a počítanie listových uzlov
-    int num_items = 0;
-    Tree* temp = root;
-    // Rekurzívne prechádzanie a počítanie
-    void count_items(Tree* tree) {
-        if (tree == NULL) return;
-        if (tree->left == NULL && tree->right == NULL) {
-            num_items++;
-        }
-        count_items(tree->left);
-        count_items(tree->right);
-    }
+    // Підрахуємо кількість всіх елементів (питань або відповідей) у дереві
+    int items_count = count_items(root);
+    printf("Pozna %d druhov ovocia a zeleniny.\n", items_count);  // Виводимо кількість елементів
 
-    count_items(root);
-    printf("Pozna %d druhov ovocia a zeleniny.\n", num_items);
+    run_system(root);  // Запускаємо систему рішень
+    destroy_tree(root);  // Знищуємо дерево після завершення роботи
 
-    // Spustenie otázok
-    ask_question(root);
-
-    // Uvoľnenie pamäte
-    destroy_tree(root);
     return 0;
 }
\ No newline at end of file