Use list comprehension to generate list