Simplify iota handling in Haskell