Update mmba.py