From d8c9839a8501819260c6073dd1bb6c70899b0195 Mon Sep 17 00:00:00 2001
From: Hazem Krimi <krimihazem1@gmail.com>
Date: Mon, 6 May 2024 19:00:53 +0100
Subject: [PATCH] Enable function related translations

---
 include/code.h | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/include/code.h b/include/code.h
index 7c0c924..79e9a03 100644
--- a/include/code.h
+++ b/include/code.h
@@ -53,6 +53,12 @@ public:
                 case PUSH:
                     file << translatePush(filename, determineSegment(vec[1]), stoi(vec[2]));
                     break;
+                case FUNCTION:
+                    file << translateFunction(vec[1], stoi(vec[2]));
+                    break;
+                case CALL:
+                    file << translateCall(vec[1], stoi(vec[2]));
+                    break;
                 case POP:
                 default:
                     file << translatePop(filename, determineSegment(vec[1]), stoi(vec[2]));
@@ -105,6 +111,9 @@ public:
                 case NOT:
                     file << translateNot();
                     break;
+                case RETURN:
+                    file << translateReturn();
+                    break;
                 default:
                     file << translateNeg();
                     break;