Adapt coloring for MPI
(OpenMP) remove unused args from generated code
(OpenMP) push color iteration up to python
(OpenMP) add MPI support