let add stmt set = match set with
    | Init -> Init
    | ToReturn set -> ToReturn (S.add stmt set)
    | ToInfinity set -> ToInfinity (S.add stmt set)