@@ -1,2 +1,5 @@
 server/key.pem
 server/cert.pem
+.license_key
+server/__pycache__/
+python_licensing/__pycache__/
@@ -1,9 +1,9 @@
 from python_licensing.python_licensing import licensed
 
-@licensed('https://localhost:5000')
+@licensed('https://127.0.0.1:5000')
 def main():
-    print("This is a licensed function.")
+    print("The body of the licensed function has been executed.")
     input("Press any key to exit.")
 
 if __name__ == '__main__':
-    main()    
+    main()
@@ -18,13 +18,13 @@ def get_activation_key():
     Get the activation key. If the key is not stored locally, ask the user to enter it.
     Store the key locally for future use.
     """
-    key_file = os.path.join(os.path.expanduser('~'), '.license_key')
+    key_file = os.path.join(os.path.expanduser('.'), '.license_key')
     if os.path.exists(key_file):
-        with open(key_file, 'r') as f:
+        with open(key_file, 'r', encoding='utf-8') as f:
             return f.read().strip()
     else:
         key = input("Enter your activation key: ")
-        with open(key_file, 'w') as f:
+        with open(key_file, 'w', encoding='utf-8') as f:
             f.write(key)
         return key
 
@@ -38,13 +38,13 @@ def wrapper(*args, **kwargs):
             activation_key = get_activation_key()
             try:
                 response = requests.get(
-                    server_url,
+                    server_url + '/check_license',
                     params={
                         'hash': unique_hash,
                         'key': activation_key
                         },
-                        verify=False,
-                        timeout=20)
+                    verify=False,
+                    timeout=20)
                 if response.status_code == 200 and response.json()['valid']:
                     return func(*args, **kwargs)
             except requests.exceptions.Timeout:
 
@@ -41,9 +41,10 @@ def check_license(license_hash, key):
         conn_ = connect_to_db()
 
     cur = conn_.cursor()
-    cur.execute("SELECT * FROM licenses WHERE hash = %s AND activation_key = %s",
-                (license_hash, key))
+    cur.execute("SELECT * FROM licenses WHERE activation_key = %s",
+                (key,))
     result = cur.fetchone()
+    print(key, result)
     if result is None:
         return False
     return result[0]
@@ -62,7 +63,7 @@ def handle_check_license():
     and 'message' (a string with a message about the license status).
     """
     license_hash = request.args.get('hash')
-    key = request.args.get('activation_key')
+    key = request.args.get('key')
     valid = check_license(license_hash, key)
     return {'valid': valid}
 
 
@@ -53,9 +53,9 @@ def initialize_db(conn):
             cur.execute("""
                 CREATE TABLE IF NOT EXISTS licenses (
                     id SERIAL PRIMARY KEY,
-                    hash TEXT NOT NULL,
+                    hash TEXT NOT NULL UNIQUE,
                     expires_after TIMESTAMP NOT NULL,
-                    activation_key TEXT NOT NULL,
+                    activation_key TEXT NOT NULL UNIQUE,
                     activated_on TIMESTAMP
                 )
             """)
@@ -73,7 +73,7 @@ def add_entry(conn, entry):
             """,
             (entry['id'],
              entry['hash'],
-             datetime.now() + timedelta(seconds=int(entry['expires_after'])),
+             entry['expires_after'],
              entry['activation_key'],
              entry['activated_on'])
         )
@@ -115,7 +115,7 @@ def main():
     parser.add_argument('--add',
                         metavar='ENTRY',
                         help='''Add an entry to the database.
-                        Example: --add "id=1 hash=hash expires_after=1234
+                        Example: --add "id=1 hash=hash expires_after=2024-02-01
                         activation_key=activation-key activated_on=2024-01-01"''')
     parser.add_argument('--remove',
                         metavar='ID',