Update mohseni_reg_all.py