diff --git a/angular.treeview.js b/angular.treeview.js
index 4b23b95..a77ba0f 100644
--- a/angular.treeview.js
+++ b/angular.treeview.js
@@ -43,6 +43,8 @@
 
 				//children
 				var nodeChildren = attrs.nodeChildren || 'children';
+                
+                var nodeCustomImage = attrs.nodeCustomImage || 'customImage';
 
 				//tree template
 				var template =
@@ -50,7 +52,8 @@
 						'<li data-ng-repeat="node in ' + treeModel + '">' +
 							'<i class="collapsed" data-ng-show="node.' + nodeChildren + '.length && node.collapsed" data-ng-click="' + treeId + '.selectNodeHead(node)"></i>' +
 							'<i class="expanded" data-ng-show="node.' + nodeChildren + '.length && !node.collapsed" data-ng-click="' + treeId + '.selectNodeHead(node)"></i>' +
-							'<i class="normal" data-ng-hide="node.' + nodeChildren + '.length"></i> ' +
+							'<i class="normal" data-ng-hide="(node.' + nodeChildren + '.length) || (node.' + nodeCustomImage + ')"></i> ' +
+                            '<i class="custom" data-ng-hide="(node.' + nodeChildren + '.length)" ng-hide="!(node.' + nodeCustomImage + ')" style="background-image:{{node.' + nodeCustomImage + '}}"></i>' +
 							'<span data-ng-class="node.selected" data-ng-click="' + treeId + '.selectNodeLabel(node)">{{node.' + nodeLabel + '}}</span>' +
 							'<div data-ng-hide="node.collapsed" data-tree-id="' + treeId + '" data-tree-model="node.' + nodeChildren + '" data-node-id=' + nodeId + ' data-node-label=' + nodeLabel + ' data-node-children=' + nodeChildren + '></div>' +
 						'</li>' +
@@ -71,6 +74,8 @@
 
 							//Collapse or Expand
 							selectedNode.collapsed = !selectedNode.collapsed;
+                            
+                            scope[treeId].currentNodeHead = selectedNode;
 						};
 
 						//if node label clicks,
diff --git a/angular.treeview.min.js b/angular.treeview.min.js
index e02a85b..4f71526 100644
--- a/angular.treeview.min.js
+++ b/angular.treeview.min.js
@@ -4,6 +4,4 @@
 	License: MIT
 */
 
-(function(f){f.module("angularTreeview",[]).directive("treeModel",function($compile){return{restrict:"A",link:function(b,h,c){var a=c.treeId,g=c.treeModel,e=c.nodeLabel||"label",d=c.nodeChildren||"children",e='<ul><li data-ng-repeat="node in '+g+'"><i class="collapsed" data-ng-show="node.'+d+'.length && node.collapsed" data-ng-click="'+a+'.selectNodeHead(node)"></i><i class="expanded" data-ng-show="node.'+d+'.length && !node.collapsed" data-ng-click="'+a+'.selectNodeHead(node)"></i><i class="normal" data-ng-hide="node.'+
-d+'.length"></i> <span data-ng-class="node.selected" data-ng-click="'+a+'.selectNodeLabel(node)">{{node.'+e+'}}</span><div data-ng-hide="node.collapsed" data-tree-id="'+a+'" data-tree-model="node.'+d+'" data-node-id='+(c.nodeId||"id")+" data-node-label="+e+" data-node-children="+d+"></div></li></ul>";a&&g&&(c.angularTreeview&&(b[a]=b[a]||{},b[a].selectNodeHead=b[a].selectNodeHead||function(a){a.collapsed=!a.collapsed},b[a].selectNodeLabel=b[a].selectNodeLabel||function(c){b[a].currentNode&&b[a].currentNode.selected&&
-(b[a].currentNode.selected=void 0);c.selected="selected";b[a].currentNode=c}),h.html('').append($compile(e)(b)))}}})})(angular);
\ No newline at end of file
+!function(a){"use strict";a.module("angularTreeview",[]).directive("treeModel",["$compile",function(a){return{restrict:"A",link:function(b,c,d){var e=d.treeId,f=d.treeModel,g=d.nodeId||"id",h=d.nodeLabel||"label",i=d.nodeChildren||"children",j=d.nodeCustomImage||"customImage",k='<ul><li data-ng-repeat="node in '+f+'"><i class="collapsed" data-ng-show="node.'+i+'.length && node.collapsed" data-ng-click="'+e+'.selectNodeHead(node)"></i><i class="expanded" data-ng-show="node.'+i+'.length && !node.collapsed" data-ng-click="'+e+'.selectNodeHead(node)"></i><i class="normal" data-ng-hide="(node.'+i+".length) || (node."+j+')"></i> <i class="custom" data-ng-hide="(node.'+i+'.length)" ng-hide="!(node.'+j+')" style="background-image:{{node.'+j+'}}"></i><span data-ng-class="node.selected" data-ng-click="'+e+'.selectNodeLabel(node)">{{node.'+h+'}}</span><div data-ng-hide="node.collapsed" data-tree-id="'+e+'" data-tree-model="node.'+i+'" data-node-id='+g+" data-node-label="+h+" data-node-children="+i+"></div></li></ul>";e&&f&&(d.angularTreeview&&(b[e]=b[e]||{},b[e].selectNodeHead=b[e].selectNodeHead||function(a){a.collapsed=!a.collapsed,b[e].currentNodeHead=a},b[e].selectNodeLabel=b[e].selectNodeLabel||function(a){b[e].currentNode&&b[e].currentNode.selected&&(b[e].currentNode.selected=void 0),a.selected="selected",b[e].currentNode=a}),c.html("").append(a(k)(b)))}}}])}(angular);
\ No newline at end of file
diff --git a/css/angular.treeview.css b/css/angular.treeview.css
index 8c94009..19e3c96 100644
--- a/css/angular.treeview.css
+++ b/css/angular.treeview.css
@@ -45,6 +45,11 @@ div[data-tree-model] li .normal {
   background-repeat: no-repeat;
 }
 
+div[data-tree-model] li .custom {
+    padding: 1px 10px;
+    background-repeat: no-repeat;
+}
+
 div[data-tree-model] li i, div[data-tree-model] li span {
   cursor: pointer;
 }