changed to list comprehension because yeah