Cleaner list comprehension as per suggestion.