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;