diff --git a/cv7/program.c b/cv7/program.c
index 2fffa63..e4ad0d1 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -1,122 +1,121 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <assert.h>
 
 #define SIZE 256
 
-// Структура вузла дерева
-struct TreeNode {
-    char *value;            // текст питання або відповіді
-    struct TreeNode *left;   // "так" гілка
-    struct TreeNode *right;  // "ні" гілка
-};
+// Štruktúra pre uzol binárneho stromu
+typedef struct TreeNode {
+    char value[SIZE];         // Otázka alebo odpoveď
+    struct TreeNode* yes;     // Ukazovateľ na "áno" vetvu
+    struct TreeNode* no;      // Ukazovateľ na "nie" vetvu
+} TreeNode;
 
-// Функція для створення вузла дерева з текстом
-struct TreeNode* createNode(const char *text) {
-    struct TreeNode *node = (struct TreeNode*) malloc(sizeof(struct TreeNode));
-    node->value = strdup(text);
-    node->left = NULL;
-    node->right = NULL;
-    return node;
-}
-
-// Функція для завантаження бази знань у форматі pre-order
-struct TreeNode* loadTree(FILE *file) {
+// Funkcia na načítanie stromu z pravidiel v súbore
+TreeNode* read_tree(FILE* file) {
     char buffer[SIZE];
-    if (!fgets(buffer, SIZE, file) || buffer[0] == '\n') {
-        return NULL;
+    if (fgets(buffer, SIZE, file) == NULL || buffer[0] == '\n') {
+        return NULL;  // Koniec súboru alebo prázdny riadok
     }
 
-    // Створюємо вузол для поточного рядка
-    struct TreeNode *node = createNode(buffer);
+    // Vytvoríme nový uzol a skopírujeme hodnotu
+    TreeNode* node = (TreeNode*)malloc(sizeof(TreeNode));
+    strncpy(node->value, buffer, SIZE);
 
-    // Якщо це відповідь (починається з '*'), то повертаємо листовий вузол
+    // Ak je to odpoveď (začína hviezdičkou), uzol je listový
     if (buffer[0] == '*') {
-        return node;
+        node->yes = NULL;
+        node->no = NULL;
+    } else {
+        // Inak načítame rekurzívne "áno" a "nie" podstromy
+        node->yes = read_tree(file);
+        node->no = read_tree(file);
     }
-
-    // Інакше рекурсивно завантажуємо лівий і правий нащадки
-    node->left = loadTree(file);
-    node->right = loadTree(file);
     return node;
 }
 
-// Рекурсивна функція для підрахунку товарів у базі знань
-int countItems(struct TreeNode *node) {
-    if (node == NULL) {
-        return 0;
-    }
-    if (node->left == NULL && node->right == NULL) {
-        return 1; // Листовий вузол є товаром
-    }
-    return countItems(node->left) + countItems(node->right);
-}
-
-// Функція для запуску експертної системи з питаннями і відповідями
-void runExpertSystem(struct TreeNode *node) {
+// Funkcia na uvoľnenie stromu z pamäte
+void destroy_tree(TreeNode* node) {
     if (node == NULL) return;
-
-    // Виводимо поточне питання або відповідь
-    printf("%s", node->value);
-
-    // Якщо це листовий вузол, то закінчуємо роботу
-    if (node->left == NULL && node->right == NULL) {
-        printf("Koniec\n");
-        return;
-    }
-
-    // Отримуємо відповідь користувача
-    char answer;
-    scanf(" %c", &answer);
-
-    // Переходимо до відповідного нащадка
-    if (answer == 'a') {
-        runExpertSystem(node->left);
-    } else if (answer == 'n') {
-        runExpertSystem(node->right);
-    } else {
-        printf("Neplatna odpoved. Koniec programu.\n");
-    }
-}
-
-// Функція для очищення пам'яті
-void freeTree(struct TreeNode *node) {
-    if (node == NULL) return;
-    freeTree(node->left);
-    freeTree(node->right);
-    free(node->value);
+    destroy_tree(node->yes);
+    destroy_tree(node->no);
     free(node);
 }
 
-int main() {
-    // Відкриваємо файл з базою знань
-    FILE *file = fopen("data.txt", "r");
-    if (file == NULL) {
-        printf("Не вдалося відкрити файл бази знань.\n");
-        return 0;
+// Funkcia na výpis stromu s odsadením (na ladenie)
+void print_tree(TreeNode* node, int offset) {
+    if (node == NULL) return;
+
+    for (int i = 0; i < offset; i++) {
+        printf(" ");
+    }
+    printf("%s", node->value);
+
+    // Rekurzívne vypíšeme "áno" a "nie" podstromy
+    if (node->yes) print_tree(node->yes, offset + 3);
+    if (node->no) print_tree(node->no, offset + 3);
+}
+
+// Funkcia na spočítanie listových uzlov (tovarov)
+int count_leaves(TreeNode* node) {
+    if (node == NULL) return 0;
+    if (node->yes == NULL && node->no == NULL) return 1;  // Je to listový uzol
+    return count_leaves(node->yes) + count_leaves(node->no);
+}
+
+// Funkcia na spustenie znalostného systému
+void run_system(TreeNode* node) {
+    if (node == NULL) return;
+
+    // Ak je uzol listový (odpoveď), vypíšeme odpoveď a skončíme
+    if (node->yes == NULL && node->no == NULL) {
+        printf("MUDrC to vie.\n");
+        printf("%s", node->value);
+        return;
     }
 
-    // Завантажуємо базу знань в дерево
-    struct TreeNode *root = loadTree(file);
+    // Zobrazíme otázku a čakáme na vstup používateľa
+    printf("%s", node->value);
+    char answer;
+    scanf(" %c", &answer);
+
+    // Podľa odpovede pokračujeme v "áno" alebo "nie" vetve
+    if (answer == 'a') {
+        run_system(node->yes);
+    } else if (answer == 'n') {
+        run_system(node->no);
+    } else {
+        printf("Odpovedajte 'a' alebo 'n'\n");
+    }
+}
+
+// Hlavná funkcia
+int main() {
+    FILE* file = fopen("baza.txt", "r");
+    if (!file) {
+        fprintf(stderr, "Nepodarilo sa otvoriť súbor\n");
+        return 1;
+    }
+
+    // Načítanie stromu z pravidiel v súbore
+    TreeNode* root = read_tree(file);
     fclose(file);
 
-    // Перевірка наявності порожнього рядка після бази знань
+    // Skontrolujeme, či je súbor správne načítaný
     if (root == NULL) {
-        printf("Помилка: База знань некоректна або пуста.\n");
-        return 0;
+        printf("Chyba: neplatná báza pravidiel\n");
+        return 1;
     }
 
-    // Рахуємо кількість товарів і виводимо повідомлення
-    int itemCount = countItems(root);
-    printf("Expert z bufetu to vie.\n");
-    printf("Pozna %d druhov ovocia a zeleniny.\n", itemCount);
-    printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+    // Spočítame listové uzly (tovary)
+    int leaves_count = count_leaves(root);
+    printf("Pozná %d druhov ovocia a zeleniny.\n", leaves_count);
 
-    // Запуск експертної системи
-    runExpertSystem(root);
-
-    // Звільняємо пам'ять
-    freeTree(root);
+    // Spustíme znalostný systém
+    run_system(root);
 
+    // Uvoľníme pamäť
+    destroy_tree(root);
     return 0;
 }