diff --git a/.gitignore b/.gitignore
index 229571f..79811b6 100644
--- a/.gitignore
+++ b/.gitignore
@@ -12,4 +12,4 @@ project/plugins/project/
 
 # intellij specific
 .idea/
-
+*.iml
\ No newline at end of file
diff --git a/src/main/scala/com/rainysoft/decisiontree/tree/Tree.scala b/src/main/scala/com/rainysoft/decisiontree/tree/Tree.scala
index 7b0f59e..a715c9b 100644
--- a/src/main/scala/com/rainysoft/decisiontree/tree/Tree.scala
+++ b/src/main/scala/com/rainysoft/decisiontree/tree/Tree.scala
@@ -1,6 +1,5 @@
 package com.rainysoft.decisiontree.tree
 
-import scala.collection.Traversable
 import scala.collection.immutable.Map
 
 /** A tree is either a leaf with a classification value