diff --git a/examples/util/mod.rs b/examples/util/mod.rs
index 25fdc67..6b5ba2d 100644
--- a/examples/util/mod.rs
+++ b/examples/util/mod.rs
@@ -1,6 +1,6 @@
 pub mod event;
 
-use tui_tree_widget::{flatten, identifier, TreeItem, TreeState};
+use tui_tree_widget::{flatten, get_identifier_without_leaf, TreeItem, TreeState};
 
 pub struct StatefulTree<'a> {
     pub state: TreeState,
@@ -52,7 +52,7 @@ impl<'a> StatefulTree<'a> {
     pub fn close(&mut self) {
         let selected = self.state.selected();
         if !self.state.close(&selected) {
-            let (head, _) = identifier::get_without_leaf(&selected);
+            let (head, _) = get_identifier_without_leaf(&selected);
             self.state.select(head);
         }
     }
diff --git a/src/lib.rs b/src/lib.rs
index 96c2a60..d3525c4 100755
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,5 @@
 #![allow(clippy::must_use_candidate)]
 
-use crate::identifier::{TreeIdentifier, TreeIdentifierVec};
 use std::collections::HashSet;
 use tui::buffer::Buffer;
 use tui::layout::{Corner, Rect};
@@ -9,10 +8,13 @@ use tui::text::Text;
 use tui::widgets::{Block, StatefulWidget, Widget};
 use unicode_width::UnicodeWidthStr;
 
-pub mod flatten;
-pub mod identifier;
+mod flatten;
+mod identifier;
 
-pub use self::flatten::flatten;
+pub use flatten::{flatten, Flattened};
+pub use identifier::{
+    get_without_leaf as get_identifier_without_leaf, TreeIdentifier, TreeIdentifierVec,
+};
 
 #[derive(Debug, Default, Clone)]
 pub struct TreeState {