diff --git a/src/main/c/Calculator/calculator.c b/src/main/c/Calculator/calculator.c
index a25009c..5ea7709 100644
--- a/src/main/c/Calculator/calculator.c
+++ b/src/main/c/Calculator/calculator.c
@@ -48,4 +48,9 @@ switch (choice) {
                 break;
             default:
                 printf("Invalid choice\n");
-        }
\ No newline at end of file
+        }
+                }
+    } while (choice != 5);
+
+    return 0;
+}