set function in python example