Add check on number of MPI tasks along axes