def METHOD_NAME(self, node):
    self.generic_visit(node)
    if node not in self.loops_mod:
        return node
    headers = [h for h, _ in self.loops_mod[node]]
    steps = [s for _, s in self.loops_mod[node]]
    node.body = steps + node.body
    return headers + [node]