Customize the icons on the js tree

I want to add a folder open close icon for closing and expanding, and a leaf icon for leaf nodes.

please help

Edit :

Tried adding the type plugin, but it doesn't seem to work.

  var data = {
      'core': {
          'data': dataObj
      },
      "search": {
          "case_insensitive": true,
          "show_only_matches": true
      },
      "plugins": ["search", "themes"]
  };
  $('#jstree_category').jstree(data);
  $('#jstree_category').on("loaded.jstree", function (e, data) {
      _this.treeLoaded = true;
      if (!_this.dataFetched) {
          return;
      }
  });
   // bind customize icon change function in jsTree open_node event.
  $('#jstree_category').on('open_node.jstree', function (e, data) {
      $('#' + data.node.id).find('i.jstree-icon.jstree-themeicon').first()
          .removeClass('glyphicon glyphicon-folder-close').addClass('glyphicon glyphicon-folder-open');
      $('#' + data.node.id).find('i.jstree-icon.jstree-themeicon')
          .addClass('glyphicon glyphicon-leaf');
  });
   // bind customize icon change function in jsTree close_node event.
  $('#jstree_category').on('close_node.jstree', function (e, data) {
      $('#' + data.node.id).find('i.jstree-icon.jstree-themeicon').first()
          .removeClass('glyphicon glyphicon-folder-open').addClass('glyphicon glyphicon-folder-close');
      $('#' + data.node.id).find('i.jstree-icon.jstree-themeicon')
          .addClass('glyphicon glyphicon-leaf');
  });