diff --git a/lib/imbecile/grammar.rb b/lib/imbecile/grammar.rb
index 11b1d2d..ae03cad 100644
--- a/lib/imbecile/grammar.rb
+++ b/lib/imbecile/grammar.rb
@@ -16,6 +16,8 @@ module Imbecile
           # Skip comment lines.
         elsif line =~ /^\s*$/
           # Skip blank lines.
+        elsif line =~ /^\s*class\s+(\S+)$/
+          @class = $1
         elsif line =~ /^\s*token\s+(\S+)\s+(.*)$/
           name, expr = $1, $2
           unless name =~ /^[a-zA-Z_][a-zA-Z_0-9]*$/