buggy, working to integrate the notebook and control flow in simple_ui