Use list comprehensions