diff --git a/calculator.c b/calculator.c
index 832dfd2..c7cfc7d 100644
--- a/calculator.c
+++ b/calculator.c
@@ -4,8 +4,9 @@ void add(int a, int b)
 {
     printf("a + b = %d\n", a + b);
 }
-void sub()
+void sub(int a, int b)
 {
+    printf("a - b = %d\n", a - b);
 }
 void mul()
 {
@@ -40,4 +41,4 @@ int main(void)
         }
     }
     return 0;
-}
\ No newline at end of file
+}