1 files changed, 20 insertions, 0 deletions
diff --git a/src/ClassLoader/EnhancedClassLoader.js b/src/ClassLoader/EnhancedClassLoader.js
new file mode 100644
index 0000000..cc4c667
--- /dev/null
+++ b/src/ClassLoader/EnhancedClassLoader.js
@@ -0,0 +1,20 @@
+import DefaultClassLoader from "./DefaultClassLoader";
+import EnhancedPassword from "../Model/Password/EnhancedPassword";
+import EnhancedFolder from "../Model/Folder/EnhancedFolder";
+import EnhancedTag from "../Model/Tag/EnhancedTag";
+
+export default class EnhancedClassLoader extends DefaultClassLoader {
+    /**
+     *
+     * @return {Object}
+     * @protected
+     */
+    _getDefaultClasses() {
+        let classes = super._getDefaultClasses();
+        classes['model.password'] = (d) => { return new EnhancedPassword(d, this.getInstance('client')); };
+        classes['model.folder'] = (d) => { return new EnhancedFolder(d, this.getInstance('client')); };
+        classes['model.tag'] = (d) => { return new EnhancedTag(d, this.getInstance('client')); };
+
+        return classes;
+    }
+}
+\ No newline at end of file