Update pyarchinit_matrix_exp.py