diff --git a/awsume/awsumepy/lib/aws.py b/awsume/awsumepy/lib/aws.py
index 0665d3c..524d6ba 100644
--- a/awsume/awsumepy/lib/aws.py
+++ b/awsume/awsumepy/lib/aws.py
@@ -1,21 +1,16 @@
-from __future__ import annotations
-
 import os
-import argparse
-import dateutil
-import json
+from typing import Union
 
 import boto3
 import botocore
 import colorama
-from datetime import datetime
+import dateutil
 
-from . import profile as profile_lib
 from . import cache as cache_lib
-from . exceptions import RoleAuthenticationError, UserAuthenticationError
-from . logger import logger
-from . safe_print import safe_print
-
+from . import profile as profile_lib
+from .exceptions import RoleAuthenticationError, UserAuthenticationError
+from .logger import logger
+from .safe_print import safe_print
 
 DEFAULT_REGION = 'us-east-1'
 
@@ -42,7 +37,7 @@ def assume_role(
     role_duration: int = None,
     mfa_serial: str = None,
     mfa_token: str = None,
-    tags: list | None = None,
+    tags: Union[list, None] = None,
 ) -> dict:
     if len(session_name) < 2:
         session_name = session_name.center(2, '_')