diff --git a/Algo/B2/Exercises/Imperative exercise.md b/Algo/B2/Exercises/Imperative exercise.md
index 6cac977..161112a 100644
--- a/Algo/B2/Exercises/Imperative exercise.md	
+++ b/Algo/B2/Exercises/Imperative exercise.md	
@@ -38,4 +38,19 @@ def max3 (x: int, y: int, z: int) -> int:
 		m = y
 	elif ((z > x) and (z > y)):
 		m = z
-	return m
\ No newline at end of file
+	return m
+
+
+def main():
+	print("Give a 3-digit number")
+	n = int(input())
+	if test(n):
+		res = loop(n)
+		if res != 1:
+			print(res, "is the mystery number")
+		else:
+			print("Try again")
+			main()
+	else:
+		print("set a 3 digits number")
+		main()
\ No newline at end of file