Update mmba_investigate.py