Adding class return type to AST.