1 files changed, 9 insertions, 2 deletions
diff --git a/src/rules/classes.rs b/src/rules/classes.rs
index 5cd29d6..cc827f0 100644
--- a/src/rules/classes.rs
+++ b/src/rules/classes.rs
@@ -3,6 +3,7 @@ use serde::Deserialize;
 use serde_yaml;
 use std::collections::HashMap;
 use std::error::Error;
+// use std::fmt;
 use std::string::String;
 
 #[derive(Deserialize)]
@@ -31,7 +32,13 @@ impl Classes {
         Ok(Classes { classes })
     }
 
-    pub fn class(&self, class_name: &str) -> Option<&Class> {
-        self.classes.get(class_name)
+    pub fn class(&self, class_name: &str) -> Result<&Class, &'static str> {
+        self.classes.get(class_name).ok_or("Class not found.")
     }
 }
+//
+// impl fmt::Display for Class {
+//     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+//         write!(f, "{}", self.name)
+//     }
+// }