how to create a function with parameters in python