Python Program for armstrong numbers