Remove list comprehension