cleanup and reorder code;save prediction object also for single prediction

1 files changed, 6 insertions, 1 deletions

diff --git a/prediction.rb b/prediction.rb
index 4e46d52..afceb08 100644
--- a/prediction.rb
+++ b/prediction.rb
@@ -9,8 +9,9 @@ module OpenTox
     field :compound, type: BSON::ObjectId
     field :model, type: BSON::ObjectId
     field :prediction, type: Hash, default:{}
+    field :csv, type: String
 
-    attr_accessor :compound, :model, :prediction
+    attr_accessor :compound, :model, :prediction, :csv
 
     def compound
       self[:compound]
@@ -24,6 +25,10 @@ module OpenTox
       self[:prediction]
     end
 
+    def csv
+      self[:csv]
+    end
+
   end
 
 end