Add option to compile with openmp