Refactor OpenMP initialization