Use a list comprehension to save space