diff --git a/code/project.py b/code/project.py
index 35b07b8a..3914fa5e 100644
--- a/code/project.py
+++ b/code/project.py
@@ -42,5 +42,8 @@ def print_sieve(num: int) -> None:
 # Driver code
 if __name__ == '__main__':
     num = int(input("Enter a number: "))
+    while num < 2:
+        print("Limit smaller than 2, pick another number")
+        num = int(input("Enter a number: "))
     print(f"Following are the prime numbers smaller than or equal to {num}")
     print_sieve(num)
\ No newline at end of file