code: def split(self, split_on):
          
          if self.op in split_on: return list(self.args)
          else: return [ self ]
docs: Splits the AST if its operation is `split_on` (i.e., return all the arguments). Otherwise, return a list with
          just the AST.