4 Digit Armstrong Number In Python Using Functions