Cleaned up the code to generate test data

1 files changed, 7 insertions, 2 deletions

diff --git a/config.py b/config.py
index 7d01933..bff125a 100644
--- a/config.py
+++ b/config.py
@@ -5,8 +5,12 @@ class Config:
     ASSETS_DEBUG = False
 
 
-class ProductionConfig(Config):
-    pass
+class YouthConfig(Config):
+    DATABASE = "test_data_large.csv"
+
+
+class SeniorConfig(Config):
+    DATABASE = "test_data_large.csv"
 
 
 class DevelopmentConfig(Config):
@@ -14,3 +18,4 @@ class DevelopmentConfig(Config):
     DEVELOPMENT = True
     TEMPLATES_AUTO_RELOAD = True
     ASSETS_DEBUG = True
+    DATABASE = "test_data_large.csv"