Implement state copy in AST