diff --git a/problems/medium/m15.py b/problems/medium/m15.py
index ab58cd0..afcb2e3 100644
--- a/problems/medium/m15.py
+++ b/problems/medium/m15.py
@@ -1,30 +1,16 @@
 '''Write a program to check if a number is an Armstrong number (e.g., 153  = 1^3 + 5^3 + 3^3 ) .'''
-def count_digits(n):
-   i = 0
-   while n > 0:
-      n //= 10
-      i += 1
-   return i
+a=input("Enter Number:")
+def Arms(a):
+   e=0
+   for i in a:
+      c=int(i)
+      d=c*c*c
+      e+=d
+   return e
 
-def sum(n):
-   i = count_digits(n)
-   s = 0
-   temp=n
-   while temp > 0:
-      digit = temp%10
-      temp//= 10
-      s += pow(digit,i)
-   return s
-
-
-num = 1634
-
- 
-s = sum(num)
-
- 
-if s == num:
-   print('Given number is an Armstrong Number')
+b=Arms(a)
+f=int(a)
+if b==f:
+   print(b,"is an Armstrong Number")
 else:
-   print('Given number is not an Armstrong Number')
-
+   print(b,"is not An Armstrong Number")
\ No newline at end of file