diff --git a/custom-login/main.py b/custom-login/main.py
index fcaf339..0b86265 100644
--- a/custom-login/main.py
+++ b/custom-login/main.py
@@ -1,5 +1,5 @@
 import base64
-import requests
+import requests, random, string
 
 from flask import Flask, render_template, url_for, redirect, session, json, request
 from flask_login import (
diff --git a/okta-hosted-login/main.py b/okta-hosted-login/main.py
index d2f118f..558c4e1 100644
--- a/okta-hosted-login/main.py
+++ b/okta-hosted-login/main.py
@@ -1,4 +1,4 @@
-import requests
+import requests, random, string
 
 from flask import Flask, render_template, redirect, request, url_for
 from flask_login import (
diff --git a/requirements.txt b/requirements.txt
index 7627feb..5cd9105 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,8 @@
 requests==2.21.0
-Flask==1.0.2
+Flask==2.0.0
 flask-cors==3.0.10
 pyOpenSSL==23.2.0
 Flask-Login==0.4.1
-okta-jwt-verifier==0.2.0
\ No newline at end of file
+okta-jwt-verifier==0.2.0
+Werkzeug~=2.0.0
+Jinja2
\ No newline at end of file