diff --git a/src/ncsi_util.hpp b/src/ncsi_util.hpp
index 750f0f1..dfa39e5 100644
--- a/src/ncsi_util.hpp
+++ b/src/ncsi_util.hpp
@@ -58,6 +58,23 @@
  */
 int getInfo(int ifindex, int package);
 
+/* @brief  This function assigns a mask controlling responses to AEN from a
+ * package.
+ * @param[in] ifindex - Interface Index.
+ * @param[in] mask - A 32-bit mask integer
+ * @returns 0 on success and negative value for failure.
+ */
+int setPackageMask(int ifindex, unsigned int mask);
+
+/* @brief  This function sets the AEN mask for the channels inside the selected
+ * package.
+ * @param[in] ifindex - Interface Index.
+ * @param[in] package - NCSI Package.
+ * @param[in] mask - A 32-bit mask integer
+ * @returns 0 on success and negative value for failure.
+ */
+int setChannelMask(int ifindex, int package, unsigned int mask);
+
 } // namespace ncsi
 } // namespace network
 } // namespace phosphor