Added self reference to custom element to extend vm method.

package.json

@@ -1,6 +1,6 @@
 {
     "name": "@contasystemer/angularjs-custom-element",
-    "version": "1.0.1",
+    "version": "1.0.2",
     "description": "AngularJS custom element module provides functionality to wrap directive/component with custom element.",
     "main": "index.js",
     "scripts": {

src/cs-custom-element.service.js

@@ -34,6 +34,7 @@
 
         var service = {
             create: create,
+            extendVm: extendVm,
             helper: {
                 decode: decode,
             },
@@ -130,17 +131,6 @@
                  */
                 identity: _.identity,
             },
-            /**
-             * @ngdoc method
-             * @name csCustomElement#extendVm
-             *
-             * @description
-             * Extends custom element scope's `vm`.
-             *
-             * @param {Object} vm Current `vm`.
-             * @returns {Object} Extended `vm`.
-             */
-            extendVm: _.identity,
         };
 
         return service;
@@ -457,7 +447,7 @@
 
                 // Create custom element scope which is used as a store of data and as a parent for component scope.
                 self.$scope = $rootScope.$new();
-                self.$scope.vm = service.extendVm({});
+                self.$scope.vm = service.extendVm(self, {});
 
                 return self;
             };
@@ -718,6 +708,23 @@
             );
         }
 
+        // VM
+
+        /**
+         * @ngdoc method
+         * @name csCustomElement#extendVm
+         *
+         * @description
+         * Extends custom element scope's `vm`.
+         *
+         * @param {HTMLElement} self Reference to the custom element.
+         * @param {Object} vm Current `vm`.
+         * @returns {Object} Extended `vm`.
+         */
+        function extendVm(self, scope) {
+            return scope;
+        }
+
         // DECODERS
 
         /**