diff --git a/baselines/src/py_loader.rs b/baselines/src/py_loader.rs
index 9288873..b12d43c 100644
--- a/baselines/src/py_loader.rs
+++ b/baselines/src/py_loader.rs
@@ -53,11 +53,11 @@ impl PyLoader {
     }
     
     pub fn put_price(&self) -> HashMap<String, f64> {
-        self.object_stores.iter().map(|o| (o.fully_qualified_name(), o.cost.get_cost)).collect::<HashMap<_,_>>()
+        self.object_stores.iter().map(|o| (o.fully_qualified_name(), o.cost.put_cost)).collect::<HashMap<_,_>>()
     }
 
     pub fn storage_price(&self) -> HashMap<String, f64> {
-        self.object_stores.iter().map(|o| (o.fully_qualified_name(), o.cost.get_cost)).collect::<HashMap<_,_>>()
+        self.object_stores.iter().map(|o| (o.fully_qualified_name(), o.cost.size_cost)).collect::<HashMap<_,_>>()
     }
 
     pub fn network_price(&self) -> HashMap<String, HashMap<String, f64>> {
@@ -100,4 +100,12 @@ impl PyLoader {
     pub fn object_store_names(&self) -> Vec<String> {
         self.object_stores.iter().map(|o| o.fully_qualified_name()).collect::<Vec<_>>()
     }
+
+    fn __repr__(&self) -> String {
+        format!("{:?}", self)
+    }
+
+    fn __str__(&self) -> String {
+        self.__repr__()
+    }
 }
\ No newline at end of file