create function in python