diff --git a/src/controller/dataControllerv2.py b/src/controller/dataControllerv2.py
index 2bda573..360d13e 100644
--- a/src/controller/dataControllerv2.py
+++ b/src/controller/dataControllerv2.py
@@ -146,4 +146,4 @@ def _init_database_(self):
             logging.error(f'Error creating {self.database}:', e)
 
     def _is_connected_(self):
-        return self.connection is not None and self.connection.is_connected()
\ No newline at end of file
+        return self.connection is not None and self.connection.is_connected()
diff --git a/src/main.py b/src/main.py
index a20842e..b404b6f 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,8 +1,32 @@
-from controller.dataController import DataController
+import logging
+import argparse
+
+from controller.dataControllerv2 import DataController2 as DataController
 
 def main(db):
+    db.connect()
+    print("HI")
+    db.disconnect()
     print(type(db))
 
-if __name__ == "__main__":
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Run unit tests with different logging levels')
+    parser.add_argument('-D', '--debug', action='store_const', const=logging.DEBUG, dest='log_level', 
+                        help='Set logging level to DEBUG')
+    parser.add_argument('-I', '--info', action='store_const', const=logging.INFO, dest='log_level', 
+                        help='Set logging level to INFO (default)')
+    parser.add_argument('-W', '--warning', action='store_const', const=logging.WARNING,dest='log_level', 
+                        help='Set logging level to WARNING')
+    parser.add_argument('-E', '--error', action='store_const', const=logging.ERROR, dest='log_level', 
+                        help='Set logging level to ERROR')
+    parser.add_argument('-C', '--critical', action='store_const', const=logging.CRITICAL, dest='log_level',
+                        help='Set logging level to CRITICAL')
+    args = parser.parse_args()
+
+    if args.log_level is not None:
+        logging.basicConfig(level=args.log_level)
+    else:
+        logging.basicConfig(level=logging.INFO)
+
     db = DataController()
-    main(db)
\ No newline at end of file
+    main(db)