created a list comprehension, tested and works