how to create user defined function in python