use list comprehension instead of ugly for