added list comprehension for python fanciness