Allow removing control flow statements