define function in python with example