PE1, with list comprehension in Haskell