edit list comprehension