diff --git a/examples/continuous-id3.rb b/examples/continuous-id3.rb
index c980d1a..42f4432 100644
--- a/examples/continuous-id3.rb
+++ b/examples/continuous-id3.rb
@@ -6,6 +6,8 @@ include DecisionTree
 
 # Read in the training data
 training = []
+attributes = nil
+
 File.open('data/continuous-training.txt', 'r').each_line do |line|
   data = line.strip.chomp('.').split(',')
   attributes ||= data
diff --git a/examples/discrete-id3.rb b/examples/discrete-id3.rb
index ef44020..6f06c79 100644
--- a/examples/discrete-id3.rb
+++ b/examples/discrete-id3.rb
@@ -5,6 +5,8 @@ require 'decisiontree'
 
 # Read in the training data
 training = []
+attributes = nil
+
 File.open('data/discrete-training.txt', 'r').each_line do |line|
   data = line.strip.split(',')
   attributes ||= data
@@ -45,7 +47,7 @@ File.open('data/discrete-test.txt', 'r').each_line do |line|
       v
     end
   end
-  training.push(test_data)
+  test.push(test_data)
 end
 
 # Let the tree predict the output and compare it to the true specified value
diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
index 4e4c340..b0702c8 100755
--- a/lib/decisiontree/id3_tree.rb
+++ b/lib/decisiontree/id3_tree.rb
@@ -190,7 +190,7 @@ module DecisionTree
         end
         label_text = "#{key} ''"
         if type(attr[0].attribute) == :continuous
-          label_text.gsub!("''", attr[0].threshold)
+          label_text.gsub!("''", attr[0].threshold.to_s)
         end
 
         [parent_text, child_text, label_text]