Haskell solution to Chocolate Feast