diff --git a/VERSION b/VERSION
index 9e11b32..9fc80f9 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-0.3.1
+0.3.2
\ No newline at end of file
diff --git a/decisiontree.gemspec b/decisiontree.gemspec
index 2d9589e..168768a 100644
--- a/decisiontree.gemspec
+++ b/decisiontree.gemspec
@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{decisiontree}
-  s.version = "0.3.1"
+  s.version = "0.3.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Ilya Grigorik"]
@@ -46,7 +46,8 @@ Gem::Specification.new do |s|
      "examples/discrete-id3.rb",
      "examples/simple.rb"
   ]
-
+  s.add_runtime_dependency "graphr"
+  
   if s.respond_to? :specification_version then
     current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
     s.specification_version = 3
diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
index c8cce8b..2656ac1 100755
--- a/lib/decisiontree/id3_tree.rb
+++ b/lib/decisiontree/id3_tree.rb
@@ -3,11 +3,7 @@
 ### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com>
 ### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com>
 
-begin
-  require 'graph/graphviz_dot'
-rescue LoadError
-  STDERR.puts "graph/graphviz_dot not installed, graphing functionality not included."
-end
+require 'graphr'
 
 class Object
   def save_to_file(filename)