should have used list comprehension!