diff --git a/src/utils.py b/src/utils.py
index faae9e3..f88c179 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -41,7 +41,11 @@ def divide(a: float, b: float) -> float:
     Returns:
     float
     '''
-    return a / b
+    
+    if b!=0: 
+        return a / b
+    else:
+        print("b shouldn't be zero")
 
 def modulo(a: int, b: int):
     '''
@@ -91,7 +95,7 @@ def return_hexadecimal(a: int) -> float:
     return hex(a)
 
 
-def return_random_number() -> int:
+def return_random_number(a: int, b:int) -> int:
     '''
     ...
 
@@ -102,5 +106,4 @@ def return_random_number() -> int:
     Returns:
     float
     '''
-
-    return np.random.randint(0, 100)
\ No newline at end of file
+    return np.random.randint(a, b)