Add openmp to elemnwise