Python Define Function Input