Use list comprehension