diff --git a/models/main.go b/models/main.go
index ba61d15..45ac3b8 100644
--- a/models/main.go
+++ b/models/main.go
@@ -14,7 +14,15 @@ const (
 func StrToWordColor(s string) WordColor {
 	switch s {
 	case "white":
-		// finish this switch; if not found/default should be beije; ai!
+		return WordColorWhite
+	case "blue":
+		return WordColorBlue
+	case "red":
+		return WordColorRed
+	case "black":
+		return WordColorBlack
+	default:
+		return "beige"
 	}
 }