Make a function in python