diff --git a/src/main.rs b/src/main.rs
index 2b02b3d..a600c54 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,16 +1,22 @@
-use language::lexer::{Lexer, Token};
+use language::lexer::{Lexer, TokenKind};
 
 fn main() {
     let input = r#"
-    var a;
-    a = 1 + 2;
+    # This is a comment
+
+    var var1 = 1;
+    var1 = var1 + 1;
+
     function add(a, b) {
         return a + b;
     }
+
+    const var2 = 2;
+    const var3 = add(var1, var2);
     "#;
 
     let mut lexer = Lexer::new(input);
-    let tokens: Vec<Token> = lexer.by_ref().collect();
+    let tokens: Vec<TokenKind> = lexer.by_ref().collect();
 
     for token in tokens {
         println!("{:?}", token);