Check Armstrong Number in Python