Add AST re-write rules