1 files changed, 3 insertions, 2 deletions
diff --git a/model.py b/model.py
index f768d0f..c652ce8 100644
--- a/model.py
+++ b/model.py
@@ -1,3 +1,4 @@
+from os import getenv
 from math import floor
 from csv import reader
 from typing import Tuple
@@ -30,7 +31,7 @@ class Game:
 
 @dataclass
 class CourtList:
-    total: int
+    total: int = int(getenv("COURTS"))
 
     def __post_init__(self) -> None:
         self.courts = {i + 1: True for i in range(self.total)}
@@ -54,7 +55,7 @@ class CourtList:
 
 @dataclass
 class PlayerList:
-    location: str
+    location: str = getenv("DATABASE")
     guests: int = 0
 
     def __post_init__(self) -> None: