playing with list comprehension a bit