Added parallel support with mpi4py