how to write a function with variable number of arguments in python