Armstrong Number Using Function In Python