changes to list comprehension