Modularise your program using function. - Another example, a function to test Armstrong Number