Update Armstrong_number.py
diff --git a/Armstrong_number.py b/Armstrong_number.py
index 0c680c7def..be923c0bf3 100644
--- a/Armstrong_number.py
+++ b/Armstrong_number.py
@@ -1,18 +1,21 @@
 def is_armstrong_number(number):
-    sum = 0
+    total = 0
 
     # find the sum of the cube of each digit
     temp = number
     while temp > 0:
-       digit = temp % 10
-       sum += digit ** 3
-       temp //= 10
+        digit = temp % 10
+        total += digit ** 3
+        temp //= 10
     
-    # display the result
-    if number == sum:
-       print(number,"is an Armstrong number")
+    # return the result
+    if number == total:
+        return True
     else:
-       print(number,"is not an Armstrong number")
+        return False
 
-number = int(input("Enter the number : "))
-is_armstrong_number(number)
+number = int(input("Enter the number: "))
+if is_armstrong_number(number):
+    print(number,"is an Armstrong number")
+else:
+    print(number,"is not an Armstrong number")

The variable name sum is a built-in function name in Python. It is not a good practice to use built-in function names as variable names. It is recommended to use a different variable name, such as total or sum_of_cubes.

The code is using the print function to display the result. It is generally better to return the result instead of printing it, so that the function can be used in other parts of the code.