1 files changed, 8 insertions, 2 deletions
diff --git a/src/runMain/kotlin/Semantic.kt b/src/runMain/kotlin/Semantic.kt
index 38dbd38..b903f84 100644
--- a/src/runMain/kotlin/Semantic.kt
+++ b/src/runMain/kotlin/Semantic.kt
@@ -6,11 +6,17 @@ class Semantic {
      * TODO: Handle scopes via { and }
      */
     fun check(statements: MutableList<MutableList<Token>>): Boolean {
-        val variables = mutableListOf<List<Token>>()
+        val variables = mutableListOf<String>()
         statements.forEach { statement ->
             if (statement[0].type == Variable) {
                 if (nextNonEmpty(statement, 0).type == Assignment) {
-                    variables.add(statement.take(5))
+                    variables.add(statement[0].content)
+                }
+            }
+
+            for (token in statement) {
+                if (token.type == Variable && token.content !in variables) {
+                    throw Exception("Undeclared variable")
                 }
             }
         }