Updated list comprehension