Modify OpenMP code: Add the module