Use list comprehension instead