diff --git a/tests/test_main.py b/tests/test_main.py
index 495a43a..21ddf3d 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,23 +1,12 @@
-# Import what we need for testing
-import unittest
-from main import app
 
-class TestApp(unittest.TestCase):
-    
-    def test_index(self):
-        # Create a test client using the Flask app context
-        with app.test_client() as client:
-            # Make a GET request to the index route
-            response = client.get('/')
-            # Check that the response status code is 200 OK
-            self.assertEqual(response.status_code, 200)
-    
-    def test_cow(self):
-        # Create a test client using the Flask app context
-        with app.test_client() as client:
-            # Make a GET request to the cow route
-            response = client.get('/cow')
-            # Check that the response status code is 200 OK
-            self.assertEqual(response.status_code, 200)
-            # Check that the response data is the expected string
-            # self.assertEqual(response.data.decode('utf-8'), 'MOoooOo!')
+def add_numbers(a, b):
+    return a + b
+
+
+def main():
+    result = add_numbers(2, 3)
+    print(result)
+
+
+if __name__ == '__main__':
+    main()