add a list comprehension