write a program to check armstrong number in python