Defining a Function in Python