Haskell Function Definition Python