Update mohseni_reg.py