diff --git a/src/yang_schema.hpp b/src/yang_schema.hpp
index 51c3be0..d100302 100644
--- a/src/yang_schema.hpp
+++ b/src/yang_schema.hpp
@@ -8,6 +8,7 @@
 
 #pragma once
 
+#include <functional>
 #include <set>
 #include <stdexcept>
 #include <unordered_map>
@@ -40,6 +41,11 @@
     yang::LeafDataTypes leafType(const path_& location, const ModuleNodePair& node) const override;
     std::set<std::string> childNodes(const path_& path) const override;
 
+    void registerModuleCallback(const std::function<std::string(const char*, const char*, const char*)>& clb);
+
+    /** @short Loads a module called moduleName. */
+    void loadModule(const std::string& moduleName);
+
     /** @short Adds a new module passed as a YANG string. */
     void addSchemaString(const char* schema);