Using Haskell Arrow in Python