Update Haskell code