Python Define Function Input Parameter