list comprehension modification