program using function in python