diff --git a/shuttleai/__init__.py b/shuttleai/__init__.py
index c541f7b..aa9c0ad 100644
--- a/shuttleai/__init__.py
+++ b/shuttleai/__init__.py
@@ -1,7 +1,6 @@
 __title__ = "shuttleai"
 __version__ = "4.1.7"
 
-
 import json
 import time
 import typing
@@ -16,49 +15,59 @@
 CACHE_FILE = Path(f"{__title__}-version.json")
 CACHE_DURATION = 86400  # 24 hours in seconds
 
-def read_cached_version_info() -> dict[str, typing.Any] | None:
+def read_cached_version_info() -> typing.Optional[dict[str, typing.Any]]:
+    """Read the cached version information from the cache file."""
     if CACHE_FILE.exists():
         try:
             with open(CACHE_FILE, "r") as file:
                 return json.load(file)  # type: ignore
-        except Exception as e:
+        except (json.JSONDecodeError, IOError) as e:
             print(f"Error reading cache file: {e}")
     return None
 
-def write_cached_version_info(version_info: dict) -> None:
+def write_cached_version_info(version_info: dict[str, typing.Any]) -> None:
+    """Write the version information to the cache file."""
     try:
         with open(CACHE_FILE, "w") as file:
             json.dump(version_info, file)
-    except Exception as e:
+    except IOError as e:
         print(f"Error writing to cache file: {e}")
 
+def is_cache_valid(cache_time: float) -> bool:
+    """Check if the cache is still valid based on the cache duration."""
+    return time.time() - cache_time < CACHE_DURATION
+
 def check_for_updates() -> None:
+    """Check for updates and notify the user if a newer version is available."""
     cached_version_info = read_cached_version_info()
 
     if cached_version_info:
         cache_time = cached_version_info.get("time")
-        if cache_time and time.time() - cache_time < CACHE_DURATION:
+        if cache_time and is_cache_valid(cache_time):
             latest_version = cached_version_info.get("version")
-            if latest_version:
-                if version.parse(__version__) < version.parse(latest_version):
-                    print(f"WARNING: You are using an outdated version of {__title__} ({__version__}). "
-                          f"The latest version is {latest_version}. It is recommended to upgrade using:\n"
-                          f">> pip install -U {__title__}")
-                return
+            if latest_version and version.parse(__version__) < version.parse(latest_version):
+                print_update_message(latest_version)
+            return
 
     try:
         response = requests.get("https://pypi.org/pypi/shuttleai/json")
+        response.raise_for_status()
         latest_version = response.json()["info"]["version"]
         write_cached_version_info({"version": latest_version, "time": time.time()})
-
         if version.parse(__version__) < version.parse(latest_version):
-            print(f"WARNING: You are using an outdated version of {__title__} ({__version__}). "
-                  f"The latest version is {latest_version}. It is recommended to upgrade using:\n"
-                  f">> pip install -U {__title__}")
-    except Exception as e:
+            print_update_message(latest_version)
+    except requests.RequestException as e:
         print(f"Could not check for updates: {e}")
 
+def print_update_message(latest_version: str) -> None:
+    """Print a message to the user indicating that an update is available."""
+    print(f"WARNING: You are using an outdated version of {__title__} ({__version__}). "
+          f"The latest version is {latest_version}. It is recommended to upgrade using:\n"
+          f">> pip install -U {__title__}")
+
+
 _patch_httpx()
 check_for_updates()
 
+
 __all__ = ["ShuttleAI", "AsyncShuttleAI"]