diff --git a/PWGEM/Dilepton/CMakeLists.txt b/PWGEM/Dilepton/CMakeLists.txt
index 7ff980baa4f..a36a3f0190a 100644
--- a/PWGEM/Dilepton/CMakeLists.txt
+++ b/PWGEM/Dilepton/CMakeLists.txt
@@ -13,4 +13,4 @@ add_subdirectory(Core)
 # add_subdirectory(DataModel)
 add_subdirectory(Tasks)
 add_subdirectory(TableProducer)
-
+add_subdirectory(Utils)
diff --git a/PWGEM/Dilepton/Utils/CMakeLists.txt b/PWGEM/Dilepton/Utils/CMakeLists.txt
new file mode 100644
index 00000000000..6a57e353eda
--- /dev/null
+++ b/PWGEM/Dilepton/Utils/CMakeLists.txt
@@ -0,0 +1,14 @@
+# Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+# See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+# All rights not expressly granted are reserved.
+#
+# This software is distributed under the terms of the GNU General Public
+# License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+#
+# In applying this license CERN does not waive the privileges and immunities
+# granted to it by virtue of its status as an Intergovernmental Organization
+# or submit itself to any jurisdiction.
+
+o2physics_add_library(EmDileptonMlCore
+  SOURCES EmDileptonMlResponse.cxx
+  PUBLIC_LINK_LIBRARIES O2Physics::MLCore)
diff --git a/PWGEM/Dilepton/Utils/ElectronModule.h b/PWGEM/Dilepton/Utils/ElectronModule.h
index fddcd6fe56c..d90769b69a4 100644
--- a/PWGEM/Dilepton/Utils/ElectronModule.h
+++ b/PWGEM/Dilepton/Utils/ElectronModule.h
@@ -9,6 +9,7 @@
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 
+/// \file ElectronModule.h
 /// \brief write relevant information about primary electrons.
 /// \author daiki.sekihata@cern.ch
 
@@ -51,7 +52,6 @@
 #include <Framework/HistogramSpec.h>
 #include <Framework/InitContext.h>
 #include <Framework/OutputObjHeader.h>
-#include <Framework/runDataProcessing.h>
 #include <MathUtils/Utils.h>
 #include <PID/PIDTOFParamService.h>
 #include <ReconstructionDataFormats/DCA.h>
diff --git a/PWGEM/Dilepton/Utils/EmDileptonMlResponse.cxx b/PWGEM/Dilepton/Utils/EmDileptonMlResponse.cxx
new file mode 100644
index 00000000000..4dc3b4b32a0
--- /dev/null
+++ b/PWGEM/Dilepton/Utils/EmDileptonMlResponse.cxx
@@ -0,0 +1,22 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+/// \file EmDileptonMlResponse.cxx
+/// \brief Helper file to provide compilation command for headers.
+///
+/// \author Vít Kučera <vit.kucera@cern.ch>, Inha University
+
+#include "ElectronModule.h"                  // IWYU pragma: keep
+#include "MlResponseDielectronSingleTrack.h" // IWYU pragma: keep
+#include "MlResponseFwdTrack.h"              // IWYU pragma: keep
+#include "MlResponseO2Track.h"               // IWYU pragma: keep
+#include "MlResponsePID.h"                   // IWYU pragma: keep
+#include "MlResponseSCT.h"                   // IWYU pragma: keep