Add OpenMP version. TODO: mix with MPI