Work on AST: Add class AST