switched to list comprehension