Make petsc import first in __init__.py

This way, we ensure that we get to import petsc4py and initialise it
appropriately.