diff --git a/njvm.c b/njvm.c
index b2bd442..652144e 100644
--- a/njvm.c
+++ b/njvm.c
@@ -238,6 +238,12 @@ int main(int argc, char *argv[]) {
             } else if (strcmp(argv[i], "--help") == 0) {
                 help();
                 return 0;
+            } else if (strcmp(argv[i], "--stack") == 0) {
+                i++;
+                // TODO: implement stack size
+            } else if(strcmp(argv[i], "--heap") == 0) {
+                i++;
+                // TODO: implement heap size
             } else {
                 sizeSDA = fromFile(argv[i], program);
                 run = 1;