diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
index b0702c8..0b47129 100755
--- a/lib/decisiontree/id3_tree.rb
+++ b/lib/decisiontree/id3_tree.rb
@@ -133,8 +133,8 @@ def graph(filename, file_type = 'png')
       dgp.write_to_file("#{filename}.#{file_type}", file_type)
     end
 
-    def ruleset
-      rs = Ruleset.new(@attributes, @data, @default, @type)
+    def rule_set
+      rs = RuleSet.new(@attributes, @data, @default, @type)
       rs.rules = build_rules
       rs
     end
@@ -260,8 +260,8 @@ def accuracy(data = nil)
     end
   end
 
-  class Ruleset
-    attr_accessor :rules
+  class RuleSet
+    attr_accessor :rules, :train_data
 
     def initialize(attributes, data, default, type)
       @attributes = attributes
@@ -322,7 +322,7 @@ def initialize(attributes, data, default, type)
 
     def train(data = @data, attributes = @attributes, default = @default)
       @classifiers = []
-      10.times { @classifiers << Ruleset.new(attributes, data, default, @type) }
+      10.times { @classifiers << RuleSet.new(attributes, data, default, @type) }
       @classifiers.each do |c|
         c.train(data, attributes, default)
       end