be more idiomatic - use list comprehension instead :)