Modified for changes to jstree directive