diff --git a/du5/program.c b/du5/program.c
index 4f4559c..312aa4e 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -121,7 +121,7 @@ int main(void) {
     char buf[128];
     while (cur) {
         if (cur->is_answer) {
-            printf("* %s\n", cur->text);
+            printf("*%s\n", cur->text);
             break;
         }
         printf("%s\n", cur->text);
@@ -138,8 +138,12 @@ int main(void) {
         }
         if (c == 'a' || c == 'A') {
             cur = cur->yes;
+            printf("Koniec\n");
+            break;
         } else if (c == 'n' || c == 'N') {
             cur = cur->no;
+            printf("Koniec\n");
+            break;
         } else {
             printf("Nerozumiem\n");
             break;