Update control_flow.ipynb from Billy