Add function for removing conditional back-edges