From 4af7886159faf3e69658114e42428089d78f3166 Mon Sep 17 00:00:00 2001
From: TurinTech Bot <bot@turintech.ai>
Date: Tue, 16 Jul 2024 10:12:32 +0000
Subject: [PATCH] Artemis Changes

---
 src/algorithms/primes.cc | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/algorithms/primes.cc b/src/algorithms/primes.cc
index d4250ab..224782b 100644
--- a/src/algorithms/primes.cc
+++ b/src/algorithms/primes.cc
@@ -1,16 +1,23 @@
+#include <cmath>
 #include "primes.h"
 
 /** @brief Checks if a number is prime
  *
  *  @param n Number to check
  *  @return True if n is prime, false otherwise
- */
+ */ 
 bool
 Primes::IsPrime(int n) {
   if (n <= 1) {
     return false;
   }
-  for (int i = 2; i < n; i += 1) {
+  if (n == 2) {
+    return true;
+  }
+  if (n % 2 == 0) {
+    return false;
+  }
+  for (int i = 3; i * i <= n; i += 2) {
     if (n % i == 0) {
       return false;
     }
@@ -51,4 +58,4 @@ Primes::PrimeFactors(int n) {
     }
   }
   return factors;
-}
+}
\ No newline at end of file