Update petsc.py