Improve haskell highlighting