Haskell Function Definition Python Return