4 Digit Armstrong Number In Python Using Return