Hadn't used a list comprehension before. Nice tip