diff --git a/index.ts b/index.ts
index 166a12ba..6c8619cb 100644
--- a/index.ts
+++ b/index.ts
@@ -9,7 +9,7 @@ import {
 
 import { Tree } from './src/tree';
 
-import { NodeMenuItemAction } from './src/menu/menu.events';
+import { NodeMenuItemAction, NodeMenuEvent } from './src/menu/menu.events';
 import { NodeMenuItem } from './src/menu/node-menu.component';
 
 import {
@@ -22,10 +22,15 @@ import {
   NodeExpandedEvent,
   NodeCollapsedEvent,
   MenuItemSelectedEvent,
-  NodeDestructiveEvent
+  NodeDestructiveEvent,
+  NodeUncheckedEvent,
+  NodeCheckedEvent,
+  NodeIndeterminedEvent,
+  NodeUnselectedEvent
 } from './src/tree.events';
 
 import { TreeComponent } from './src/tree.component';
+import { TreeController } from './src/tree-controller';
 import { TreeModule } from './src/tree.module';
 
 export {
@@ -44,10 +49,16 @@ export {
   NodeExpandedEvent,
   NodeCollapsedEvent,
   NodeDestructiveEvent,
+  NodeMenuEvent,
+  NodeUncheckedEvent,
+  NodeCheckedEvent,
+  NodeIndeterminedEvent,
+  NodeUnselectedEvent,
   TreeComponent,
   TreeModule,
   NodeMenuItemAction,
   NodeMenuItem,
   ChildrenLoadingFunction,
-  MenuItemSelectedEvent
+  MenuItemSelectedEvent,
+  TreeController
 };
diff --git a/package.json b/package.json
index cee74e0e..598abb54 100644
--- a/package.json
+++ b/package.json
@@ -33,8 +33,7 @@
   "scripts": {
     "compile": "ngc -p tsconfig-aot.json && node umd-bundler.js",
     "clean": "rimraf coverage build dist dist-demo bundles factories .publish",
-    "pre:publish":
-      "npm run clean && npm run test && npm run lint && npm run compile && mkdir -p dist && cp -R src/styles.css README.md media dist",
+    "pre:publish": "npm run clean && npm run test && npm run lint && npm run compile && mkdir -p dist && cp -R src/styles.css README.md media dist",
     "post:publish": "npm run build:demo && gh-pages -d dist-demo",
     "start": "ng serve",
     "build:demo": "ng build",