diff --git a/models/state.go b/models/state.go
index a7a8bb0..1dd2733 100644
--- a/models/state.go
+++ b/models/state.go
@@ -1,26 +1,36 @@
 package models
 
-type UserTeam string
+type (
+	UserTeam string
+	UserRole string
+)
 
 const (
+	// UserTeam
 	UserTeamBlue = "blue"
 	UserTeamRed  = "red"
 	UserTeamNone = "none"
+	//UserRole
+	UserRoleMime    = "mime"
+	UserRoleGuesser = "guesser"
+	UserRoleNone    = "none"
 )
 
-func StrToUserTeam(s string) WordColor {
+func StrToUserTeam(s string) UserTeam {
 	switch s {
 	case "blue":
 		return UserTeamBlue
 	case "red":
 		return UserTeamRed
-	case "none":
+	default:
 		return UserTeamNone
 	}
-	return WordColorUserTeamNone
 }
 
+// write StrToUserRole func; ai!
+
 type UserState struct {
 	Username string
 	Room     RoomPublic
+	Team     UserTeam
 }