diff --git a/lab-02/problem-04.py b/lab-02/problem-04.py
index 672cb04..f73fbfc 100644
--- a/lab-02/problem-04.py
+++ b/lab-02/problem-04.py
@@ -1,11 +1,15 @@
 # Write a Python program that prompts the user to input two numbers and raises a TypeError exception if the inputs are
 # not numerical.
 
-def numb_check(x, y):
-    try:
-        result = x * y
-    except TypeError:
-        print(f"Either {x} and/or {y} are not numbers.")
+def numb_check(n):
+    while True:
+        try:
+            numb = float(input(n))
+        except ValueError:
+            print(f"{numb} is not a number")
 
 
-numb_check(x = input("Enter a number: "), y = input("Enter another number: "))
\ No newline at end of file
+x = numb_check(input('Enter a number: '))
+y = numb_check(input('Enter another number: '))
+product = x * y
+print(product)
\ No newline at end of file